package com.xxxx.config;

import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.xxxx.pojo.MailConstants;
import com.xxxx.pojo.MailLog;
import com.xxxx.service.IMailLogService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.amqp.core.Binding;
import org.springframework.amqp.core.BindingBuilder;
import org.springframework.amqp.core.DirectExchange;
import org.springframework.amqp.core.Queue;
import org.springframework.amqp.rabbit.connection.CachingConnectionFactory;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.Collection;
import java.util.Iterator;

/**
 * @program: yun2
 * @ClassName RabbitConfig
 * @description:
 * @author: ZhouBin
 * @create: 2022-06-04 17:10
 * @Version 1.0
 **/
@Configuration
public class RabbitConfig {
    private static final Logger LOGGER= LoggerFactory.getLogger(RabbitConfig.class);
    @Autowired
    private CachingConnectionFactory cachingConnectionFactory;
    @Autowired
    private IMailLogService mailLogService;

    @Bean
    public RabbitTemplate rabbitTemplate(){
        RabbitTemplate rabbitTemplate = new RabbitTemplate(cachingConnectionFactory);
        rabbitTemplate().setConfirmCallback((data,ack,cause) ->{
            String msgId = data.getId();
            if (ack){
                LOGGER.info("{}=========>消息发送成功",msgId);
                mailLogService.update(new UpdateWrapper<MailLog>().set("status",1).eq("msgId",msgId));
            }else {
                LOGGER.info("{}=========>消息发送失败",msgId);
            }

        });
        rabbitTemplate().setReturnCallback((msg,repCode,repText,exchange,routingkey) -> {
            LOGGER.info("{}=========>消息发送成功",msg.getBody());

        });
        return rabbitTemplate;
    }





    @Bean
    public Queue queue(){
        return new Queue(MailConstants.MAIL_QUEUE_NAME);

    }

    @Bean
    public DirectExchange directExchange(){
        return new DirectExchange(MailConstants.MAIL_EXCHANGE_NAME);
    }

    @Bean
    public Binding binding(){
        return BindingBuilder.bind(queue()).to(directExchange()).with(MailConstants.MAIL_ROUTING_KEY_NAME);
    }
}
